package sh.orm;

import sh.base.aop.util.ReflectUtil;
import sh.orm.convert.JdbcType;
import sh.orm.convert.TypeConvertor;
import sh.orm.convert.TypeConvertorRegistry;
import sh.orm.util.NameUtil;
import sh.orm.util.ResultSetUtil;

import java.lang.reflect.Field;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * entity持久化
 * ResultSet转换为Entity
 * @author lqq
 * @date 2019/1/22
 */
public class ConvertContext {

    /** 当前转换时，返回的结果集中，列名和 jdbc类型的map ，<b>只有当转换的对象不是Entity时，这个对象才有只<b/> */
    private Map<String, JdbcType> columnNameJdbcTypeMap;

    /**
     * 把结果集转化为需要的对象
     * @param rs
     * @param clazz
     * @param <T>
     * @return
     */
    public <T>  T convertToEntity(ResultSet rs, Class<T> clazz){
        List<T> dataList = convertToEntityList(rs,clazz);
        if(dataList == null || dataList.size()  <= 0){
            return null;
        }
        return dataList.get(0);
    }


    /**
     * 查询的结果是一个数字
     * @param rs
     * @return
     */
    public int convertToInt(ResultSet rs){
        try {
            //光标移动到第一行
            rs.first();

            return rs.getInt(1);
        } catch (SQLException e) {
            e.printStackTrace();
            throw  new OrmException("处理结果集异常" + e.getMessage());
        }
    }

    /**
     * 转换为一个list
     * @param rs
     * @param clazz
     * @param <T>
     * @return
     */
    public <T> List<T> convertToEntityList(ResultSet rs, Class<T> clazz){

        int rowCount = 0;
        List<T> dataList = null;
        try {
            // 将光标移动到最后一行
            rs.last();
            // 得到当前行号，即结果集记录数
            rowCount = rs.getRow();
            if(rowCount < 1){
                return null;
            }
            //光标移动到第一行
            rs.first();
            dataList = new ArrayList<>();
            for(int i = 0 ; i < rowCount ; i++){
                //把ResultSet 转换为 Map
                Map<String,Object> fieldValueMap = ResultSetUtil.convertResultSet2Map(
                        StaticEntityHolder.getStaticEntity(clazz).getColumnFieldMap(),rs);

                //如果有关联关系，返回代理对象
                T entity;
                if(EntityUtil.hasRelationAnnotation(clazz)){
                    entity = ProxyFactory.getRelationProxy(ReflectUtil.getInstance(clazz), clazz);
                }else{
                    entity = (T)ReflectUtil.getInstance(clazz);
                }

                //递归设置所有的属性值
                setFieldValue(clazz,entity,rs);

                dataList.add(entity);
                rs.next();
            }


        } catch (SQLException e) {
            e.printStackTrace();
            throw  new OrmException("处理结果集异常" + e.getMessage());
        }

        return dataList;
    }



    /**
     *
     * @param entity 携带数据的entity
     * @param resultSet 当前行的结果集
     * @param <T>
     */
    private <T> void setFieldValue(Class<T> clazz ,T entity,ResultSet resultSet){
        Field[] fields = clazz.getDeclaredFields();
        Object value ;
        for(Field field : fields){
            //transient 不设置值
            if((field.getModifiers() & 128) == 128){
                continue;
            }
            field.setAccessible(true);
            //把resultset中的类型转换为java中的类型
            String columnName = NameUtil.convertHumpToUnderline(field.getName());
            // 因为有的数据不是entity，而是其的返回类型，所以如果不是entity，
            // 需要根据当前resultset的数据ResultSet的数据类型判断jdbc数据类型，当然只需要判断一次即可
            JdbcType jdbcType ;
            //1.如果是Entity，那么从 {@link sh.orm.DatabaseMetadata }中获取数据
            //     *  2.如果是普通的java对象，需要根据结果集判断返回值的对象
            if(EntityUtil.isEntity(entity.getClass())){
                jdbcType = EntityUtil.getJdbcType(entity.getClass(), columnName);
            }else{
                //    普通的java对象
                if(columnNameJdbcTypeMap == null){
                    this.columnNameJdbcTypeMap = ResultSetUtil.getColumnNameJdbcTypeMap(resultSet);
                }
                jdbcType = columnNameJdbcTypeMap.get(columnName);
            }
            if(jdbcType == null){
                throw new OrmException("没有找到结果集对应的jdbc类型,列名：" + columnName);
            }

            TypeConvertor<T> typeConvertor = TypeConvertorRegistry.getInstance().getTypeConvertor(field.getType(), jdbcType);
            value = typeConvertor.convertToJavaType(columnName , resultSet);
            ReflectUtil.setFieldValue(field,entity,value);
        }

        if(!"Object".equals(clazz.getSuperclass().getSimpleName())){
            setFieldValue(clazz.getSuperclass(),entity,resultSet);
        }
    }


//   TODO 设置值的时候通过ResultSet获取内容

}
